package com.sun.filesigndemo.common.utils;

import org.apache.commons.io.IOUtils;
import org.springframework.core.io.DefaultResourceLoader;
import org.springframework.core.io.Resource;
import org.springframework.core.io.ResourceLoader;
import org.yaml.snakeyaml.Yaml;

import java.io.IOException;
import java.io.InputStream;
import java.util.*;

public class PropertiesLoader {

    private static ResourceLoader resourceLoader = new DefaultResourceLoader();

    private final static String SPRING_PROFILES_ACTIVE= "spring.profiles.active";

    private static String SPRING_PROFILES_ACTIVE_VALUE= System.getProperty(SPRING_PROFILES_ACTIVE) == null ?
            System.getenv(SPRING_PROFILES_ACTIVE) : System.getProperty(SPRING_PROFILES_ACTIVE);

    private final Map properties;

    public PropertiesLoader(String... resourcesPaths) {
        properties = loadProperties(resourcesPaths);
    }

    public Map getProperties() {
        return properties;
    }

    /**
     * 取出Property，但以System的Property优先,取不到返回空字符串.
     */
    private String getValue(String key) {
        String systemProperty = System.getProperty(key);
        if (systemProperty != null) {
            return systemProperty;
        }
        if (properties.containsKey(key)) {
            return properties.get(key).toString();
        }
        return "";
    }

    /**
     * 取出String类型的Property，但以System的Property优先,如果都为Null则抛出异常.
     */
    public String getProperty(String key) {
        String value = getValue(key);
        if (value == null) {
            throw new NoSuchElementException();
        }
        return value;
    }

    /**
     * 取出String类型的Property，但以System的Property优先.如果都为Null则返回Default值.
     */
    public String getProperty(String key, String defaultValue) {
        String value = getValue(key);
        return value != null ? value : defaultValue;
    }

    /**
     * 取出Integer类型的Property，但以System的Property优先.如果都为Null或内容错误则抛出异常.
     */
    public Integer getInteger(String key) {
        String value = getValue(key);
        if (value == null) {
            throw new NoSuchElementException();
        }
        return Integer.valueOf(value);
    }

    /**
     * 取出Integer类型的Property，但以System的Property优先.如果都为Null则返回Default值，如果内容错误则抛出异常
     */
    public Integer getInteger(String key, Integer defaultValue) {
        String value = getValue(key);
        return value != null ? Integer.valueOf(value) : defaultValue;
    }

    /**
     * 取出Double类型的Property，但以System的Property优先.如果都为Null或内容错误则抛出异常.
     */
    public Double getDouble(String key) {
        String value = getValue(key);
        if (value == null) {
            throw new NoSuchElementException();
        }
        return Double.valueOf(value);
    }

    /**
     * 取出Double类型的Property，但以System的Property优先.如果都为Null则返回Default值，如果内容错误则抛出异常
     */
    public Double getDouble(String key, Integer defaultValue) {
        String value = getValue(key);
        return value != null ? Double.valueOf(value) : defaultValue;
    }

    /**
     * 取出Double类型的Property，但以System的Property优先.如果都为Null或内容错误则抛出异常.
     */
    public Long getLong(String key) {
        String value = getValue(key);
        if (value == null) {
            throw new NoSuchElementException();
        }
        return Long.valueOf(value);
    }

    /**
     * 取出Double类型的Property，但以System的Property优先.如果都为Null则返回Default值，如果内容错误则抛出异常
     */
    public Long getLong(String key, Integer defaultValue) {
        String value = getValue(key);
        return value != null ? Long.valueOf(value) : defaultValue;
    }

    /**
     * 取出Boolean类型的Property，但以System的Property优先.如果都为Null抛出异常,如果内容不是true/false则返回false.
     */
    public Boolean getBoolean(String key) {
        String value = getValue(key);
        if (value == null) {
            throw new NoSuchElementException();
        }
        return Boolean.valueOf(value);
    }

    /**
     * 取出Boolean类型的Property，但以System的Property优先.如果都为Null则返回Default值,如果内容不为true/false则返回false.
     */
    public Boolean getBoolean(String key, boolean defaultValue) {
        String value = getValue(key);
        return value != null ? Boolean.valueOf(value) : defaultValue;
    }

    /**
     * 载入多个文件, 文件路径使用Spring Resource格式.
     */
    private Map loadProperties(String... resourcesPaths) {
        Yaml yaml = new Yaml();
        Map resultMap = new HashMap<>(150);
        for (String location : resourcesPaths) {

            InputStream is = null;
            try {
                Resource resource = resourceLoader.getResource(location);
                is = resource.getInputStream();
                Iterable<Object> result = yaml.loadAll(is);
                Iterator iterator = result.iterator();
                // 保存第一个分组的公有配置
                if (iterator.hasNext()) {
                    resultMap.putAll(getMap((LinkedHashMap) iterator.next(), ""));
                } else {
                    continue;
                }
                // 没有启动配置spring.profiles.active，读配首个节点中的配置spring.profiles.active
                String springProfilesActive= SPRING_PROFILES_ACTIVE_VALUE;
                if (SPRING_PROFILES_ACTIVE_VALUE == null) {
                    Object _springProfilesActive= resultMap.get("spring.profiles.active");
                    if (_springProfilesActive != null) {
                        springProfilesActive= _springProfilesActive.toString();
                    }
                }

                while (iterator.hasNext()) {
                    LinkedHashMap linkMap = (LinkedHashMap) iterator.next();

                    // 获取配置中的 spring.profiles
                    String springProfiles = "";
                    if (linkMap.get("spring") !=null && ((Map)linkMap.get("spring")).get("profiles") != null) {
                        springProfiles= ((Map)linkMap.get("spring")).get("profiles").toString();
                    }
                    // 环境变量为空 使用spring.profiles空的配置， 或者环境变量有值使用spring.profiles相等的配置
                    if ((springProfilesActive == null && springProfiles == "")
                            || springProfiles.equals(springProfilesActive)) {
                        resultMap.putAll(getMap(linkMap, ""));
                    }
                }

                if (result instanceof Map) {
                    resultMap.putAll((Map) result);
                }
            } catch (IOException ex) {
//				logger.info("Could not load properties from path:" + location + ", " + ex.getMessage());
            } finally {
                IOUtils.closeQuietly(is);
            }
        }
        return resultMap;
    }


    private static Map getMap(LinkedHashMap linkMap, String keyStr) {
        Map resultMap = new HashMap<>();

        Set keys= linkMap.keySet();
        for (Iterator key = keys.iterator(); key.hasNext();) {
            Object k = (Object) key.next();
            StringBuilder keySb= new StringBuilder(keyStr);
            keySb.append(k.toString());
            Object object= linkMap.get(k);
            if (object instanceof LinkedHashMap) {
                Map childrenMap= getMap((LinkedHashMap)object, keySb.append(".").toString());
                resultMap.putAll(childrenMap);
            } else {
                resultMap.put(keySb.toString(), linkMap.get(k));
            }

        }

        return resultMap;
    }
}
